package UDP;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;

//实现一个简单的 UDP 服务器
public class EchoServer {
    //创建 Socket 对象
    private DatagramSocket serverSocket;

    //构造方法
    public EchoServer(int port) throws Exception {
        serverSocket = new DatagramSocket(port);
    }

    //启动服务器
    public void start() throws IOException {
        System.out.println("服务器已启动，等待客户端连接...");
        while (true) {
            // 1. 读取请求并解析
            // 1) 创建一个空白的 DatagramPacket 对象
            DatagramPacket requestPacket = new DatagramPacket(new byte[4096], 4096);
            // 2) 通过 receive() 方法接收客户端发送的请求
            //    如果网卡没有接收到数据，receive() 方法会阻塞
            //    如果网卡接收到数据，receive() 方法会将数据存入 DatagramPacket 对象中
            serverSocket.receive(requestPacket);
            // 3) 把 DatagramPacket 对象中的数据解析成字符串
            String request = new String(requestPacket.getData(), 0, requestPacket.getLength());
            System.out.println("收到客户端请求：" + request);
            // 2.根据请求计算响应
            String response = process(request);
            System.out.println("响应内容：" + response);
            // 3.把响应写回客户端
            // 1) 创建一个 DatagramPacket 对象，用于封装响应数据
            DatagramPacket responsePacket = new DatagramPacket(response.getBytes(), response.getBytes().length, requestPacket.getSocketAddress());
            // 2) 通过 send() 方法发送响应数据到客户端
            serverSocket.send(responsePacket);
            // 4.打印日志
            System.out.printf("[%s:%d] req: %s, resp: %s\n", 
                requestPacket.getAddress(), requestPacket.getPort(), request, response);
        }
    }

    //这是一个回显服务器，它会将客户端发送的消息原样返回给客户端。
    public String process(String request) {
        return request;
    }

    public static void main(String[] args) throws Exception {
        EchoServer server = new EchoServer(9090);
        server.start();
    }
}
